def info_print():
    print('请选择功能--------------')
    print('1、添加学员')
    print('2、删除学员')
    print('3、修改学员')
    print('4、查询学员')
    print('5、显示所有学员')
    print('6、退出系统')
    print('-' * 20)


# 定义学生信息（因为都会涉及到修改，所以定义为全局变量）
info = []


def add_student():
    """添加学员函数"""
    """添加学员函数"""
    # 1. 用户输入：学号、姓名、手机号
    new_id = input('请输入学号：')
    new_name = input('请输入姓名：')
    new_tel = input('请输入手机号：')

    global info
    # 2、判断是否已经输入了（姓名不能重复）
    for i in info:
        if new_name == i['name']:
            print('该姓名已存在')
            return

    # 3、添加数组数据
    info.append({'id': new_id, 'name': new_name, 'tel': new_tel})
    print(info)


def delete_student():
    # 1、用户输入要删除的姓名
    delete_name = input("请输入要删除的姓名：")
    # 2、判断姓名是否存在
    global info
    for i in info:
        if delete_name == i['name']:
            # 删除数据信息
            info.remove(i)
            print('删除成功')
            break
    else:
        # 循环正常结束，说明删除数据不存在
        print('该姓名不存在')
    print(info)


# print(selectNum)


def edit_student():
    """修改学生信息"""
    # 因为姓名是唯一的，所以姓名不允许修改，就让用户输入修改的手机号就行了
    # 1、用户输入要修改的姓名
    edit_name = input('请输入要修改的姓名：')
    # 2、判断姓名是否存在
    global info
    for i in info:
        if edit_name == i['name']:
            # 3、修改手机号
            i['tel'] = input('请输入新手机号：')
            print('修改成功')
            break
    else:
        # 循环正常结束，说明姓名不存在
        print('该姓名不存在')
    print(info)


def query_student_by_name():
    """根据姓名查询学生信息"""
    # 1、用户输入要查询的姓名
    query_name = input('请输入要查询的姓名：')
    # 2、判断姓名是否存在
    for i in info:
        if query_name == i['name']:
            # 3、打印信息
            print('学号：%s, 姓名：%s, 手机号：%s' % (i['id'], i['name'], i['tel']))
            break
    else:
        # 循环正常结束，说明姓名不存在
        print('该姓名不存在')


def query_student_list():
    """显示所有学员信息"""
    # 1. 打印提示字
    print('学号\t姓名\t手机号')
    # 2. 打印所有学员的数据
    for i in info:
        print(f"{i['id']}\t{i['name']}\t{i['tel']}")


while True:
    # 先打印菜单，然后根据用户输入选择功能
    info_print()
    selectNum = int(input('请输入您的选择：'))
    if selectNum == 1:
        add_student()
    elif selectNum == 2:
        delete_student()
    elif selectNum == 3:
        edit_student()
    elif selectNum == 4:
        query_student_by_name()
    elif selectNum == 5:
        query_student_list()
    elif selectNum == 6:
        choose_res= input('您确定要退出系统吗？y/n')
        if choose_res == 'y':
            print("成功退出系统，欢迎您下次继续登录系统，再见！")
            break
    else:
        print("您输入的指令错误，请重新输入")
